package com.linyaonan.leetcode.medium._820;

import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Set;

/**
 *
 * 给定一个单词列表，我们将这个列表编码成一个索引字符串 S 与一个索引列表 A。
 *
 * 例如，如果这个列表是 ["time", "me", "bell"]，我们就可以将其表示为 S = "time#bell#" 和 indexes = [0, 2, 5]。
 *
 * 对于每一个索引，我们可以通过从字符串 S 中索引的位置开始读取字符串，直到 "#" 结束，来恢复我们之前的单词列表。
 *
 * 那么成功对给定单词列表进行编码的最小字符串长度是多少呢？
 *
 *  
 *
 * 示例：
 *
 * 输入: words = ["time", "me", "bell"]
 * 输出: 10
 * 说明: S = "time#bell#" ， indexes = [0, 2, 5] 。
 *  
 *
 * 提示：
 *
 * 1 <= words.length <= 2000
 * 1 <= words[i].length <= 7
 * 每个单词都是小写字母 。
 *
 *
 * @author: Lin
 * @date: 2020/3/28
 */
public class MinimumLengthEncoding {

    public int minimumLengthEncoding(String[] words) {
        // 单个字符情况，直接返回字符长度+1
        if (words.length == 1) {
            return words[0].length() + 1;
        }
        ArrayList<String> add = new ArrayList<>();
        for (int i = 0; i < words.length; i++) {
            String word = words[i];
            boolean replace = false;
            boolean get = false;
            int getIndex = -1;
            String replaceStr = "";
            for (int j = 0; j < add.size(); j++) {
                String s = add.get(j);
                // 单词是否存在已自身为结尾的较长单词
                if (s.endsWith(word)) {
                    get = true;
                    break;
                } else if (word.endsWith(s)) {
                    // 使用较长的值替换
                    replace = true;
                    replaceStr = word;
                    getIndex = j;
                    break;
                }
            }
            if (get) {
                continue;
            }
            if (replace) {
                add.remove(getIndex);
                add.add(replaceStr);
            } else {
                add.add(word);
            }
        }
        int count = 0;
        for (String s : add) {
            // 增加字符长度
            count += s.length();
            // 增加结尾#
            count++;
        }

        return count;
    }

    public int minimumLengthEncoding2(String[] words) {
        Set<String> good = new HashSet(Arrays.asList(words));
        for (String word: words) {
            for (int k = 1; k < word.length(); ++k) {
                good.remove(word.substring(k));
            }
        }

        int ans = 0;
        for (String word: good) {
            ans += word.length() + 1;
        }
        return ans;
    }

}
